# 给你二叉树的根结点 root ，请你将它展开为一个单链表： 
# 
#  
#  展开后的单链表应该同样使用 TreeNode ，其中 right 子指针指向链表中下一个结点，而左子指针始终为 null 。 
#  展开后的单链表应该与二叉树 先序遍历 顺序相同。 
#  
# 
#  
# 
#  示例 1： 
#  
#  
# 输入：root = [1,2,5,3,4,null,6]
# 输出：[1,null,2,null,3,null,4,null,5,null,6]
#  
# 
#  示例 2： 
# 
#  
# 输入：root = []
# 输出：[]
#  
# 
#  示例 3： 
# 
#  
# 输入：root = [0]
# 输出：[0]
#  
# 
#  
# 
#  提示： 
# 
#  
#  树中结点数在范围 [0, 2000] 内 
#  -100 <= Node.val <= 100 
#  
# 
#  
# 
#  进阶：你可以使用原地算法（O(1) 额外空间）展开这棵树吗？ 
# 
#  Related Topics 栈 树 深度优先搜索 链表 二叉树 👍 1763 👎 0
from binarytree import Node, build

from typing import Optional


# leetcode submit region begin(Prohibit modification and deletion)
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def flatten(self, root: Optional[Node]) -> None:
        """
        Do not return anything, modify root in-place instead.
        思路：每次递归都相当于把以root节点的树。把左节点都变为右节点后在连上右节点
        """

        if not root:
            return

        right_node = root.right
        if root.left:
            self.flatten(root.left)
            root.right = root.left
            root.left = None
            # self.flatten(root.left)运行完后，以root。left为根节点的树，把左节点都变为右节点后在连接上原来的右节点，已经排列好
            cur = root
            while cur.right:
                cur = cur.right

            self.flatten(right_node)
            # self.flatten(right_node)运行完后，以right_node为根节点的树，把左节点都变为右节点后在连接上原来的右节点，已经排列好
            cur.right = right_node
        else:
            # 若没有左子树则直接处理右子树
            self.flatten(right_node)

# leetcode submit region end(Prohibit modification and deletion)
# 或者从列表创建 左节点2i+1 右节点2i+2

nodes = [1, 2, 5, 3, 4, None, 6]
tree = build(nodes)
print(tree)
# 可视化打印

Solution().flatten(tree)
print(tree)
